flexible functions Python